Flatten |
您所在的位置:网站首页 › pytorch flatten函数 › Flatten |
Flatten¶
paddle.nn.
Flatten
(
start_axis=1,
stop_axis=- 1
)
[源代码]
¶
构造一个 Flatten 类的可调用对象。更多信息请参见代码示例。它实现将一个连续维度的 Tensor 展平成一维 Tensor。 参数¶start_axis (int,可选) - 展开的起始维度,默认值为 1。 stop_axis (int,可选) - 展开的结束维度,默认值为-1。 返回¶无。 代码示例¶ import paddle inp = paddle.ones([5, 2, 3, 4]).astype('float32') flatten = paddle.nn.Flatten(start_axis=1, stop_axis=2) y = flatten(inp) # y.shape = [5, 6, 4] |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |